//1337. 矩阵中战斗力最弱的 K 行
//思路：对每一行的非零元素相加，最后乘以100（因为数据范围在100以内）再加上行号，尾插到v中
//再对v进行排序，并遍历数组%100，resize(k),返回v即可


#include <vector>
#include <algorithm>
using namespace std;

class Solution {
public:
    vector<int> kWeakestRows(vector<vector<int>>& mat, int k) 
    {
        vector<int> v;
        for(int i = 0; i < mat.size(); i++)
        {
            int tmp = 0;
            for(int j = 0; j < mat[i].size(); j++)
            {
                if(mat[i][j] == 1) tmp++;
                else break;
            }
            v.push_back(tmp*100+i);
        }
        sort(v.begin(), v.end());
        for(auto& e : v)
        {
            e %= 100; 
        }
        v.resize(k);
        return v;
    }
};